package com.jim.healio.graphrag.model;

import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.AllArgsConstructor;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;
import java.util.ArrayList;
import java.util.HashMap;

@Data
@NoArgsConstructor
@AllArgsConstructor
public class MedicalDocument {
    
    /**
     * 文档唯一标识
     */
    private String id;
    
    /**
     * 文档标题
     */
    private String title;
    
    /**
     * 文档内容
     */
    private String content;
    
    /**
     * 文档类型
     */
    private DocumentType type;
    
    /**
     * 文档来源
     */
    private String source;
    
    /**
     * 文档URL或路径
     */
    private String url;
    
    /**
     * 文档作者
     */
    private String author;
    
    /**
     * 创建时间
     */
    private LocalDateTime createdAt = LocalDateTime.now();
    
    /**
     * 更新时间
     */
    private LocalDateTime updatedAt = LocalDateTime.now();
    
    /**
     * 文档元数据
     */
    private Map<String, Object> metadata = new HashMap<>();
    
    /**
     * 从文档中提取的实体ID列表
     */
    private List<String> extractedEntityIds = new ArrayList<>();
    
    /**
     * 文档标签
     */
    private List<String> tags = new ArrayList<>();
    
    /**
     * 文档章节/段落
     */
    private String section;
    
    /**
     * 文档语言
     */
    private String language = "zh-CN";
    
    /**
     * 文档版本
     */
    private String version = "1.0";
    
    /**
     * 文档状态
     */
    private DocumentStatus status = DocumentStatus.ACTIVE;
    
    /**
     * 文档类型枚举
     */
    public enum DocumentType {
        GUIDELINE("临床指南"),
        RESEARCH_PAPER("研究论文"),
        TEXTBOOK("医学教科书"), 
        DRUG_MANUAL("药物说明书"),
        CASE_STUDY("病例研究"),
        PROTOCOL("医疗协议"),
        REVIEW("综述文章"),
        REFERENCE("参考资料");
        
        private final String description;
        
        DocumentType(String description) {
            this.description = description;
        }
        
        public String getDescription() {
            return description;
        }
    }
    
    /**
     * 文档状态枚举
     */
    public enum DocumentStatus {
        ACTIVE("活跃"),
        ARCHIVED("已归档"),
        DEPRECATED("已废弃"),
        DRAFT("草稿");
        
        private final String description;
        
        DocumentStatus(String description) {
            this.description = description;
        }
        
        public String getDescription() {
            return description;
        }
    }
    
    /**
     * 便捷构造方法
     */
    public MedicalDocument(String id, String title, String content, DocumentType type) {
        this.id = id;
        this.title = title;
        this.content = content;
        this.type = type;
    }
    
    /**
     * 添加标签
     */
    public void addTag(String tag) {
        if (!this.tags.contains(tag)) {
            this.tags.add(tag);
        }
    }
    
    /**
     * 添加提取的实体
     */
    public void addExtractedEntity(String entityId) {
        if (!this.extractedEntityIds.contains(entityId)) {
            this.extractedEntityIds.add(entityId);
        }
    }
} 